more minor changes
change to the controller