Add test for caching