fix style

Chang the format of the code.