Modified controller
Update controller