change a little bit of the controller