Hao make some correction on main.cpp