Fix README to pass pytest doc test.
Merge pull request #20 from Rogdham/readme_doctest_fix

Fix README to pass pytest doc test.