Cleaning up readme and comments