working with @katur on math for momentum