It seems that Diddy has made some friends already in federal prison