U.S. judge suggests Cohen returned to prison in retaliation for Trump book