Create count.py