Updated readme and added comments