Update count.rb