Added 4 vector and invariant mass functionality