Update SU2.py