Update main.cpp

oh vi is way better than this here