Modeling of two-dimensional material-based devices