input: Write a function to generate a list of lists of words from a list of strings. The returned list should have the first word for each string followed by the following words for each string.
  def get_word_lists(strings):
      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
  
  assert get_word_lists(["Hello world", "This is a test"]) == [["Hello", ["Hello", "world"]], ["This", ["This", "is"], ["This", "is", "a"], ["This", "is", "a", "test"]]]
  assert get_word_lists(["Python code", "Function generator"]) == [["Python", ["Python", "code"]], ["Function", ["Function", "generator"]]]
  assert get_word_lists(["Single input"]) == [["Single", ["Single", "input"]]]
  assert get_word_lists([]) == []
output: def get_word_lists(strings):
      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([words[0], word_list])
      return word_lists