Update clean_distillate_streams_magnitude.py