edited driver code to comment out after the ==