adjusting code to match yellow light application