Add distribute_setup