import glob


def get_files(directory, extension):
    """
    Take a directory and an extension and return the files
    that match the extension
    """
    return glob.glob('%s/*.%s' % (directory, extension))