Add rendering of boxes and replaced balls by rectangles on the demo.