Updated reward function for keep spot task to match new control.