Update flag_cube.py