Small update in example_main_module.py