Add more doctests