added a driver class for easier readability