Update states2.py