Add count.py