Add tests in count_array_spec.rb and code to pass tests in count_array.rb