Add functions to fill yellow and red boxes