Add doctests for previous commit