Fix bugs in pick_node

Todo:
Bugs in distribute_data