An improved 2D–3D model for charge transport based on the maximum entropy principle