diff --git a/curvature_analysis.py b/curvature_analysis.py
index 8dc49c2..8a22626 100644
--- a/curvature_analysis.py
+++ b/curvature_analysis.py
@@ -133,6 +133,8 @@ def visualize_curvature_labels(folder, frame, bins, neighbourhoods):
 
     fig = make_subplots(rows=1, cols=len(colors_binned_all)+1, start_cell="top-left", specs = [ [{ 'type': 'scene' } for _ in range(len(colors_binned_all)+1)] ])
 
+    colors = np.ones((verts.shape[0],3))*np.array([255,200,200])[None,:]
+
     fig.add_trace(go.Mesh3d(
                 # 8 vertices of a cube
                 x=verts[:,0],
@@ -142,7 +144,8 @@ def visualize_curvature_labels(folder, frame, bins, neighbourhoods):
                 i = faces[:,0],
                 j = faces[:,1],
                 k = faces[:,2],
-                vertexcolor=np.ones((verts.shape[0],3))*200
+                vertexcolor=colors,
+                lighting={ 'ambient':0.9, 'diffuse':0.5, 'roughness':0.5 }
             ),
             row=1, col=1)
 
@@ -156,7 +159,8 @@ def visualize_curvature_labels(folder, frame, bins, neighbourhoods):
                     i = faces[:,0],
                     j = faces[:,1],
                     k = faces[:,2],
-                    vertexcolor=colors_binned
+                    vertexcolor=colors_binned,
+                    # lighting={ 'ambient':0.8, 'diffuse':0.5, 'roughness':0.5 }
                 ),
                 row=1, col=i+2)
     
@@ -168,9 +172,10 @@ def visualize_curvature_labels(folder, frame, bins, neighbourhoods):
                     scene4_aspectmode='manual', scene4_aspectratio=dict(x=1, y=1, z=1),
                     # scene5_aspectmode='manual', scene5_aspectratio=dict(x=1, y=1, z=1),
                     )
-    helper = dict(xaxis = dict(nticks=4, range=[-1,1],),
-                yaxis = dict(nticks=4, range=[-1,1],),
-                zaxis = dict(nticks=4, range=[-1,1],),)
+    helper = dict(xaxis = dict(range=[-1,1], showbackground=False, showticklabels=False, showaxeslabels=False, showspikes=False, visible=False),
+                yaxis = dict(range=[-1,1], showbackground=False, showticklabels=False, showaxeslabels=False, showspikes=False, visible=False),
+                zaxis = dict(range=[-1,1], showbackground=False, showticklabels=False, showaxeslabels=False, showspikes=False, visible=False),)
+        
     fig.update_layout(scene = helper,
                         scene2 = helper,
                         scene3 = helper,