Fix extractor to extract training, dev, and test data