Update main.py

remove imports no longer needed