almost finishing update_as_a_control_problem