Added '2D' option to modelling.