simplifying a bit the main.py module