Update flag.py