Removing doctests
Adding type annotation to documentation