tweak demo code style