Add docs + doctests