Updating controller to use.