Improve documentation

[Mandawi]

Documentation work using doctest https://docs.python.org/3/library/doctest.html#module-doctest