Add edit and update methods to controller.