Updated wireframe code