Update momentum.cpp