Update user controller to work