Add a decorator in utility.py