Changed main.cpp