Modification main.cpp