Updated code for wireframe