Updated main.cc