Small changes to README.md
More code comments + small tweaks