fixed the style of code