added os import

fixed error on os.remove()