fixed the layout of code