Add a flag for "decay" in momentum.