remove momentum function implemented and works
Lattice implemented, they are not given velocity at the moment