Feature extraction

Fixed integration test for feature extraction.