Update __init__.py

removed unneeded code