Class AttentionExtension