adding driver class to ease work