Made a driver class