remove code from __init__.py

Closes #56