add annotation to util.py