Array Using User Input In Python