Adding some doctests