Update useful_functions.py