Update drive_test.c