Simplify wigner-seitz code.