Add cleanup script for upgrading