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. Each sublist 
      should contain the first word of each string followed by the subsequent words.
      
      >>> 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(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