Use re.search instead of match