updated main.cpp