Move transverseMass to HasFourMomentum, add cosTheta to LorentzVector