Update count_cosine.rb