Update main.cpp

Updates