Rewrite `get_bbs` as a composite transform

[ondrasej]

The get_bbs function composes Beam pipelines. We should adopt the idiomatic Beam approach, which is to make this a composite transform.