Update retrieve_stats.py