Add documentation. Enhance behavior for running __init__.py as main