Remove code from __init__.py