Update trivial_functions.py