Python Square Все цифры в списке

def square(list):
    return [i ** 2 for i in list]
Ugliest Unicorn