The rehab system won't be coming back, because people didn't like it :P
The dungeon will be returning eventually, but I need to think of a proper reason for it to exist. It's a bit redundant to have a "discipline training" area when you can literally adjust people's minds at will, haha.
Could be used for using more potent mind control things. We already know the risks for using them in public so what if its used for that. And also maybe if theres characters that resist the mind control.