diff --git a/sheet2/C/main.py b/sheet2/C/main.py
index 2d6fa64..3be4d95 100644
--- a/sheet2/C/main.py
+++ b/sheet2/C/main.py
@@ -10,14 +10,15 @@ def u(x, p):
 
 x = np.linspace(0, 1, 200)
 
-p_values = [-10, -2, 0, 2, 10]
-colors = ['purple', 'blue', 'black', 'orange', 'red']
+t = np.linspace(-1, 1, 15)
+p_values = 50 * np.sign(t)*np.power(t,2) #make more values around 0
 
+cmap = plt.cm.rainbow
 
+plt.figure()
 
-plt.figure(figsize=(7,5))
-for p, c in zip(p_values, colors):
-    plt.plot(x, u(x, p), label=f"p = {p}", color=c, linewidth=2)
+for p, color in zip(p_values, cmap(np.linspace(0, 1, len(p_values)))):
+    plt.plot(x, u(x, p), color=color, linewidth=2, label=f"p={p:.0f}")
 
 
 plt.xlabel("x")