Update main.cpp

now should somewhat actually work