update diamond demo by 元飞