per Spark style guide, can you change this to include curly braces, e.g.
```scala
if (allLeaf) {
  break = true
} else {
  level += 1
}
```

Thanks.