Getting Array Input In Python W3schools