Updated main.cpp input