small changes
small changes on main.py