Update Driver.java

Fixed a bug (Sort of) and added some code for line comparison.