Replace re.findall() by re.match()