Update block_puzzle.py