revise main.cpp for evaluation