Clean up imports in featureset.py