Updated drive_surface.py to test