Added NMS class