Python Function Input Array