Updated test_driving_functions.m to the new imp functions