Add tests for lookup feature