Update clean-up script