Update add_eq_2.py