Update wei.py