Update main.cpp

Added stuff