[Ruby] exercise _ Element Count

Write a method element_count that takes in an array and returns a hash representing the count of each element in the array.

def element_count(arr)

end

puts element_count(["a", "b", "a", "a", "b"]) #=> {"a"=>3, "b"=>2}
puts element_count(["red", "red", "blue", "green"]) #=> {"red"=>2, "blue"=>1, "green"=>1}

my code

def element_count(arr)
  
  count = Hash.new(0)
  arr.each do |char|
  	count[char] += 1
  end
  
  
  return count

end

puts element_count(["a", "b", "a", "a", "b"]) #=> {"a"=>3, "b"=>2}
puts element_count(["red", "red", "blue", "green"]) #=> {"red"=>2, "blue"=>1, "green"=>1}

solution 

def element_count(arr)
  count = Hash.new(0)
  arr.each { |ele| count[ele] += 1 }
  return count
end

puts element_count(["a", "b", "a", "a", "b"]) #=> {"a"=>3, "b"=>2}
puts element_count(["red", "red", "blue", "green"]) #=> {"red"=>2, "blue"=>1, "green"=>1}