Add session management functionality