fixed styles
fixed main.cpp code