added functions to remove files/directory

import glob

import os

import shutil

def remove_files(fmt=""):

    """ Remove files matching the specified format. """

    for fp in glob.glob(fmt):

        os.remove(fp)

def remove_dir(path=""):

    """ Remove the specified directory. """

    shutil.rmtree(path)