Update contamination_screen_v4.py