Update __init__.py

Added doc string