input: def generate_word_lists(strings: List[str]) -> List[List[str]]:
      """Generate a list of lists of words from the strings. The returned list should have
      the first word for each string followed by following words for each string.
      >>> generate_word_lists(["Hello world", "This is a test"])
      [['Hello', 'world'], ['This', 'is', 'a', 'test']]
      """
output:     result = []
      for s in strings:
          words = s.split()
          result.append(words)
      return result