remove unnecessary import in main.py file