added driver code file and refactored tests