Clean up word_count with better regex and #each_with_object