User Array Input In Python Example