Updated README and few comments