Update __init__.py

cleanup some stuff