Linear momentum

Switching to using linear momentum rather than velocity for the magnets.