Update weighteffect.py