Add a main function in shell.py for entry point

For pip entry points I'm adding a main function for the entry point.