Added "box" figure function