Update main.py

remove suffix checking