Separate file for driver class