Update main.py

Added the regex