Diddy's 'horror' prison to get total overhaul as rapper waits for his trial