Update __init__.py

Updated header