Update count_test.R