Taking Array Input In Python