Update generate_stats.py