Added extra files to the `cleanup` function