How to modify "sweep" code for straight line drive