adjusting imports in main.py