Update main.py

Added the regex
Merge pull request #7 from Pear0/patch-2

Add Regex