update source code

for demo