Add session handling