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