Update Driver.java

Updates on the driver class for loop.java