Basic main.py rework