Removed import of shutil since it was unused