Modified so that the driver takes in argument and added README on how to run the program
added more to README