Add cleanup script