Add session management