Update "shell.py" so code lives inside "main()"