Refactored poly_features to use list comprehension.