modify demo.cpp