Create count_code.py