Add clean function