drive_test function added