Updated README, added a few comments