Update momentum.h