Adding worker versions of initial_sharding.py