Change update script and clean some things