add count.py