Move the code from __init__.py to session2.py