updated the style of the code