Removed old code from __init__.py