Add support for wireframe mode