input: Write a function to count the number of words in a string.
  assert count_words("This is a sample string") == 5
  assert count_words("Hello world") == 2
  assert count_words("Python is great") == 3
output: def count_words(s):
      return len(s.split())