Update ellipticity.py