Moved imports inside main.py