Add test for store