minor changes
update to controller