edited main.cpp