Pass Array Into Function In Python