Add UMS feature