Added Model to controller