Modify Controller generator