Exercise: Add doctests to your own code