modify main.cc