Update puzzle_cube_simulator.py