update YiZhi.py