change code style

hope it works