Update README.md

Please find comments in the code.