Finish up writing unit tests.
Update README.md