Update demo-main.cc