Python Pass Array To Function