Update README and added comments