Adds more functionality to clean script