Update get_counts.R