added driver to main method