modify stats.py