Updated README and added some code comments