Update stats.py

Bug fixes