working on generic load_distribution function