Forces user to specify idxs for flattening

This is not ideal, it would be better to calculate the idxs for the
keys, but I don't see an easy to way to do it at the momemt.