leetcode 824 goat latin

def (S):
    res = S.split()
    for i,s in enumerate(res):
        if s[0] in 'aeiouAEIOU':
            res[i] = s + 'ma' + 'a'*(i+1)
        else:
            res[i] = s[1:]+s[0] + 'ma' + 'a'*(i+1)
    return ' '.join(res)