Update common_utils.py