add grain size dependent rheology