adding in doctests