input: def get_word_lists(strings: List[str]) -> List[List[str]]:
      """Generate a list of lists of words from a list of strings. The returned list should contain
      sublists where each sublist is formed by taking the first word followed by the subsequent words
      of each string in the input list.
      >>> word_lists = get_word_lists(["Hello world", "This is a test"])
      >>> print(word_lists)
      [["Hello", "world"], ["This", "is", "a", "test"]]
      """
      word_lists = []
      for string in strings:
          words = string.split(" ")
          word_list = []
          for i in range(len(words)):
              word_list.append(words[i:])
          word_lists.append(word_list)
      return word_lists
output:     word_lists = []
      for string in strings:
          words = string.split(" ")
          word_list = [words[0]] + words[1:]
          word_lists.append(word_list)
      return word_lists