Add feature extraction section