Missed to remove changes in Controller class in the last commit. Removed them