Adding in controller generator