Update README.md
Merge pull request #1 from xuechen-sjtu/readme-edits

Readme edits, fixes #1