Update count.py