Add Controller class