Array Input In Python Using Vscode