Updated README and added comments