Update a bit of code style