add cleaning code