fix style of the code