Added main method in Driver
Modified Readme and Driver
Fixed bug