Update data generation script for ellipse regression