Add red boxes to figures