Can we merge this class extension definition into the `tree-labelable` module? I don't see a need for them to be separate modules since they're close friends :)