input: 
  def get_word_lists(strings: List[str]) -> List[List[str]]:
      """
      Generate a list of lists of words from a given list of strings. Each sublist 
      starts with the first word of the corresponding string, followed by subsequent 
      words as they appear in the string.
  
      >>> get_word_lists(["Hello world", "This is a test"])
      [['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(" ".join(words[i:]))
          word_lists.append(word_list)
      return word_lists
  
output:     word_lists = []
      for string in strings:
          words = string.split()
          word_lists.append(words)
      return word_lists