Update add_three.py