Add invariant mass function