Added momentum function