<SYSTEM_TASK:>
Solve the following problem using Python, implementing the functions described below, one line at a time
<END_TASK>
<USER_TASK:>
Description:
 def files_in_dir(path, extension):
 """Enumartes the files in path with the given extension"""

    ends = '.{0}'.format(extension)
    return (f for f in os.listdir(path) if f.endswith(ends))