User Input In Python Array