add plot_cube.py