refine utils.py