fix: updated doctest, pep8 and unit test
Merge pull request #1 from satra/fix/doctest

fix: updated doctest, pep8 and unit test