Adding functionality for Controller class