Improve output formatting in example main.py