Getting Array Input from the user using while Loop in Python