Add test for sharding