modify run2.py