include momentum in model