Add tests for first-index