Added tracking of state numbers to cts_lattice_gas.py