Consider using a trait for `callback` in the `flatten` method insead of a callback

My profiling showed that at least in some situations, those callback calls can make a difference, since those calls cannot be inlined. If the method would take a trait instead (that has a `callback` method), those calls can be inlined, which is especially beneficial for paths with many lines.