Adding a cleanup step after updating the code