more beautiful version of main.py