added smth in main.py