Improved comments in doctest