Update ez_setup.py