update lattice_opt.py