Update code style

Update code style