Напишите программу, которая заполняет массив квадратами натуральных чисел от 1 до n . (python)

Def some_func(num): 
  arr = [] 
  for n in range(num):   
    arr.append((n+1)*(n+1))
  return arr