Small main.cpp update.