updated the controller a bit