update demo & test

update demo & test