Update to follow the code style