Hi Stefan, thanks for your comments. I have made the changes and also added doctests. Ran the test suite and they all pass.