Add momentum checks into setMomentum()