<SYSTEM_TASK:>
Solve the following problem using Python, implementing the functions described below, one line at a time
<END_TASK>
<USER_TASK:>
Description:
 def rm_files(path, extension):
 """ Remove all files in the given directory with the given extension :param str path: Directory :param str extension: File type to remove :return none: """

    files = list_files(extension, path)
    for file in files:
        if file.endswith(extension):
            os.remove(os.path.join(path, file))
    return