Update to code style