Array User Input In Python