Updated controller for models