Update controller
Update test