Add session tracking