Added invariant mass function