Added count_code.py