tidy the code
fixed Main.cpp