fix few things
fix main.py