Updated main.cpp

Fixed some annoying bits