Update counting.py