how to pass a list as an input of a function in python,"print(square([1, 2, 3]))"