Made doctest more clear