step 5 - changed main.cpp