controller generator change. add argument