Add basic feature extraction functionality