Question: 
Write a function to find all words starting with 'a' or 'e' in a given string.


Model's Code:
def words_ae(s):
    return [word for word in s.split() if word.startswith(('a', 'e'))]



Sample Correct Code:
import re
def words_ae(text):
 list = re.findall("[ae]\w+", text)
 return list


Asserts:
assert words_ae("python programe")==['ame']
assert words_ae("python programe language")==['ame','anguage']
assert words_ae("assert statement")==['assert', 'atement']


Tests:
Test 1 of 3: FAIL: AssertionError()

Test 254 of 974: 181/254 (71.25984251968504%): failed