Add minor corrections to MatrixAugmented