a bit cleaning main.py file