Update simplify.py