Changes to file main.cpp