Update feature_functions.py