ENH speed up/simplify feature extraction in example