Implemented new features for plotting heterogeneous geometries