python from __future__ import print_function from builtins import range def square_numbers(n): return [x**2 for x in range(n)] if __name__ == '__main__': numbers = square_numbers(5) print(numbers)