Getting Array Input In Python 3