Added session stats handling