Integrate session into __init__