Update to Lattice.py