Add lattice vectors to kpoints

Description

Add a copy of the lattice vectors to the kpoints and use this in a few places. Also took the opportunity to clean up a bit the kpoints module.