Array Input In Function Python