Update reference_count.py