Changed name of Controller class to Model