Adding doctests
Adding more relevant documentation