edit count.py