input: from typing import List
  
  def get_word_lists(strings: List[str]) -> List[List[str]]:
      """ Generate a list of lists of words from the given list of strings.
      The returned list should have the first word for each string followed by 
      following words for each string.
      >>> word_lists = get_word_lists(["Hello world", "This is a test"])
      >>> print(word_lists) # prints "[["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 = []
          for word in words:
              word_list.append(word)
          word_lists.append(word_list)
      return word_lists