Adding weight array to load balance routines