Adding methods to the controller generator