Preliminary and final numerical models for sediment transport including grain size tracking