input: Write a python function to autocomplete words from a given prefix.
  assert auto_complete('pyth') == {'python', 'pythagorean', 'pythagoras', 'pythons'}
  assert auto_complete('pro') == {'program', 'progress', 'project'}
  assert auto_complete('exa') == {'example', 'examine', 'exaggerate'}
output: def auto_complete(prefix):
      words = {"python", "pythagorean", "pythagoras", "pythons", "program", "progress", "project", "example", "examine", "exaggerate"}
      return {word for word in words if word.startswith(prefix)}