added smth into main.py after merge with lg2