Update mem_dist.sh