Update to match code style