changed main.cpp