Added predictions in the controller