Update prison.scala