Updated example in README.

And little additions to tests.