Edited main.cpp