Update remove_noise.py