Updated README and added some comments