Added momentum class