Update code style a little bit