modelonegrain now computes the flux from each grain individually and can return them separately or summed