Add red and yellow badges