Updated README and commented code