Add BAMMtrait handling to getMarginalBranchRateMatrix.R