Fixed README.md and added comments to code