Add feature_gen