Add import in session.py