Fixed some innacuracies with extract_features(...)