Add support for shallow collection to the index