Changes to python api made while constructing the BEAVRS benchmark

This PR is for reference - the lattice stuff is a hack.