Update bkz.py