Pass Array To Function Python