fixin main.py for dropdown