Add session management functions