diff --git a/FloryHugg_DiffUnbleached.ipynb b/FloryHugg_DiffUnbleached.ipynb
index e3d4508d1d35317ac4af282a5bba435741cf7513..af443d57cadc4e979a98ad6a287ff984236e43f8 100644
--- a/FloryHugg_DiffUnbleached.ipynb
+++ b/FloryHugg_DiffUnbleached.ipynb
@@ -38,8 +38,8 @@
     "#              tc*df.inner(df.grad(c), df.grad((1-c_tot)*Ga0))+\n",
     "#              tc*df.inner(df.grad(c_tot), df.grad((1-c_tot)/c_tot*c*Ga0))) * df.dx\n",
     "#     # Weak form 1D short:\n",
-    "        form = (df.inner((c-c0)/dt, tc) +\n",
-    "                df.inner((1-c_tot)*Ga0*(df.grad(c)-c/c_tot*df.grad(c_tot)),\n",
+    "        form = ((c-c0)/dt*tc + (1-c_tot)*Ga0*\n",
+    "                df.inner((df.grad(c)-c/c_tot*df.grad(c_tot)),\n",
     "                          df.grad(tc))) * df.dx\n",
     "    elif sym == 2:\n",
     "    # Weak form radial symmetry:\n",
@@ -51,14 +51,17 @@
     "#              (1-c_tot)*Ga0*2*X[0]*c.dx(0)*tc+\n",
     "#              (1-c_tot)*Ga0/c_tot*c*2*X[0]*c_tot.dx(0)*tc) * df.dx\n",
     "#     # Weak form radial symmetry:\n",
-    "        form = ((c-c0)/dt*tc*X[0]**2 +\n",
-    "                (1-c_tot)*Ga0*(c.dx(0)-c/c_tot*c_tot.dx(0))*\n",
-    "                         tc.dx(0)*X[0]**2) * df.dx\n",
-    "    \n",
+    "#         form = ((c-c0)/dt*tc*X[0]**2 +\n",
+    "#                 (1-c_tot)*Ga0*(c.dx(0)-c/c_tot*c_tot.dx(0))*\n",
+    "#                          tc.dx(0)*X[0]**2) * df.dx\n",
+    "        form = ((c-c0)/dt*tc + (1-c_tot)*Ga0*\n",
+    "                df.inner((df.grad(c)-c/c_tot*df.grad(c_tot)),\n",
+    "                          df.grad(tc)))*X[0]*X[0]*df.dx\n",
     "    t = 0\n",
     "    # Solve in time\n",
     "    ti = time.time()\n",
     "    for i in range(n_t):\n",
+    "#         print(np.sum([x*x*c0([x]) for x in np.linspace(0, 1, 1000)]))\n",
     "        df.solve(form == 0, c)\n",
     "        df.assign(c0, c)\n",
     "        t += dt\n",