Python Array User Input In Python