clean up code, add comments, that kind of thing
Updated README.md