working on main.py