modify main cpp
modify main