diff --git a/src/my_mesh.h b/src/my_mesh.h
index e48fa14..5c6a046 100644
--- a/src/my_mesh.h
+++ b/src/my_mesh.h
@@ -10,17 +10,10 @@ using namespace OpenMesh;
 using namespace OpenMesh::Attributes;
 
 struct MyTraits : public OpenMesh::DefaultTraits {
-    // use vertex normals and vertex colors
-    VertexAttributes( OpenMesh::Attributes::Normal | OpenMesh::Attributes::Color );
-    // store the previous halfedge
-    HalfedgeAttributes( OpenMesh::Attributes::PrevHalfedge );
-    // use face normals face colors
-    FaceAttributes( OpenMesh::Attributes::Normal | OpenMesh::Attributes::Color );
-    EdgeAttributes( OpenMesh::Attributes::Color );
-    // vertex thickness
-    VertexTraits{float thickness; float value; Color faceShadingColor;};
-    // edge thickness
-    EdgeTraits{float thickness;};
+    VertexAttributes(OpenMesh::Attributes::Normal | OpenMesh::Attributes::Color);
+    HalfedgeAttributes(OpenMesh::Attributes::PrevHalfedge);
+    FaceAttributes(OpenMesh::Attributes::Normal);
+    EdgeAttributes(OpenMesh::Attributes::Color);
 };
 typedef OpenMesh::PolyMesh_ArrayKernelT<MyTraits> MyMesh;