Update flatiron.py