Updating controller

Do this?