Update README and add comments