Update main.cpp

Fixade output och lite annat