Remove the code example in favor of the doctests

What do you think about using the tests instead of the examples in the main function?