fix the layout of code