Partial Feature Extraction with minor bug