Remove unused import from main.py