Update class to UMD