fix flow logic. add walkToBall method