@@ -46,7 +46,7 @@ fig = go.Figure([
 y=[20, 14, 25, 16],
 name='Primary Product',
 # Named CSS color
-marker_color='lightblue'
+marker_color='indianred'
 )
 ])
 
@@ -95,45 +95,54 @@ supported_colors = ["aliceblue", "antiquewhite", "aqua", "aquamarine", "azure",
 "turquoise", "violet", "wheat", "white", "whitesmoke",
 "yellow", "yellowgreen"]
 
-fig = go.Figure(layout=dict(title="Supported CSS Colors in Plotly.py"))
-
-for i, color in enumerate(supported_colors):
-row, col = i // 5, i % 5
-x0, y0 = col * 1.2, -row * 1.2
+def display_colors_as_shapes(color_names):
+fig = go.Figure(layout=dict(title="Supported CSS Colors in Plotly.py"))
 
-fig.add_shape(
-type="rect",
-x0=x0, y0=y0,
-x1=x0+1, y1=y0+1,
-fillcolor=color,
-line=dict(color="black", width=0.1),
-)
+for i, color in enumerate(color_names):
+row, col = i // 5, i % 5
+x0, y0 = col * 1.2, -row * 1.2
+
+fig.add_shape(
+type="rect",
+x0=x0, y0=y0,
+x1=x0+1, y1=y0+1,
+fillcolor=color,
+line=dict(color="black", width=1),
+)
+
+fig.add_annotation(
+x=x0+0.5, y=y0-0.1,
+text=color,
+showarrow=False,
+font=dict(size=10)
+)
 
-fig.add_annotation(
-x=x0+0.5, y=y0-0.1,
-text=color,
-showarrow=False,
-font=dict(size=10)
-)
-
-fig.update_layout(
-height=((len(supported_colors) // 5) + (1 if len(supported_colors) % 5 else 0)) * 120,
-margin=dict(l=0, r=0, t=50, b=0),
-showlegend=False,
-plot_bgcolor='rgba(0,0,0,0)',
-xaxis=dict(
-showticklabels=False,
-range=[-0.5, 6]
-),
-yaxis=dict(
-showticklabels=False,
-range=[-((len(supported_colors) // 5) + 1) * 1.2, 1.5]
+height = ((len(color_names) // 5) + (1 if len(color_names) % 5 else 0)) * 120
+
+fig.update_layout(
+height=height,
+width=800,
+showlegend=False,
+plot_bgcolor='rgba(0,0,0,0)',
+margin=dict(l=50, r=50, t=50, b=50),
+xaxis=dict(
+showgrid=False,
+zeroline=False,
+showticklabels=False,
+range=[-0.5, 6]
+),
+yaxis=dict(
+showgrid=False,
+zeroline=False,
+showticklabels=False,
+scaleanchor="x",
+scaleratio=1,
+range=[-((len(color_names) // 5) + 1) * 1.2, 1.5]
+)
 )
-)
+
+return fig
 
+fig = display_colors_as_shapes(supported_colors)
 fig.show()
 ```
-
-```python
-
-```