Update controller class