Add the cleanup script