Modify controller template
Update generator