Update data_stats.py