Cleaning up readme and added some comments