Clean up README and tests

* Fix the example code in README
* Remove unused variables in tests