input: Write a python function to return a list of squares of numbers divisible by 3 from 0 to 9 using list comprehension.
  assert main() == [0, 9, 36]
output: def main():
      return [x**2 for x in range(10) if x % 3 == 0]