Add test for storage