FIX: Needed to add multiply with gridDim to give weight