fix doctest for new function signature
more PEP8-y (it's a start)