Started on the controller - have an idea to make it better