Update smooth_cube.py