added doctest lines, and expectations for tests