Doctest update