Add missing session functions