diff --git a/doc/notebooks/11_tutorial_Non_Newtonian_Flow.ipynb b/doc/notebooks/11_tutorial_Non_Newtonian_Flow.ipynb
index eabe8b35647db33fdb256c34d1381e469ba24d8c..41a1e595d25ced3d4d6c1adaba9344dcf79d4eaf 100644
--- a/doc/notebooks/11_tutorial_Non_Newtonian_Flow.ipynb
+++ b/doc/notebooks/11_tutorial_Non_Newtonian_Flow.ipynb
@@ -154,23 +154,11 @@
    "cell_type": "code",
    "execution_count": 8,
    "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/html": [
-       "<div>Subexpressions:</div><table style=\"border:none; width: 100%; \"><tr style=\"border:none\"> <td style=\"border:none\">$$vel0Term \\leftarrow {src}_{(-1,1,-1)}^{21} + {src}_{(-1,-1,-1)}^{19} + {src}_{(-1,0,-1)}^{14} + {src}_{(-1,1,0)}^{10} + {src}_{(-1,-1,0)}^{8} + {src}_{(-1,1,1)}^{25} + {src}_{(-1,-1,1)}^{23} + {src}_{(-1,0,1)}^{18} + {src}_{(-1,0,0)}^{4}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$vel1Term \\leftarrow {src}_{(1,-1,-1)}^{20} + {src}_{(0,-1,-1)}^{11} + {src}_{(1,-1,0)}^{7} + {src}_{(0,-1,0)}^{1} + {src}_{(1,-1,1)}^{24} + {src}_{(0,-1,1)}^{15}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$vel2Term \\leftarrow {src}_{(1,0,-1)}^{13} + {src}_{(1,1,-1)}^{22} + {src}_{(0,1,-1)}^{12} + {src}_{(0,0,-1)}^{5}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$\\delta_{\\rho} \\leftarrow {src}_{(0,0,0)}^{0} + {src}_{(1,0,0)}^{3} + {src}_{(1,1,0)}^{9} + {src}_{(0,1,0)}^{2} + {src}_{(1,0,1)}^{17} + {src}_{(1,1,1)}^{26} + {src}_{(0,1,1)}^{16} + {src}_{(0,0,1)}^{6} + vel0Term + vel1Term + vel2Term$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$u_{0} \\leftarrow - {src}_{(1,0,-1)}^{13} - {src}_{(1,1,-1)}^{22} - {src}_{(1,-1,-1)}^{20} - {src}_{(1,0,0)}^{3} - {src}_{(1,1,0)}^{9} - {src}_{(1,-1,0)}^{7} - {src}_{(1,0,1)}^{17} - {src}_{(1,1,1)}^{26} - {src}_{(1,-1,1)}^{24} + vel0Term + 1.21951219512195 \\cdot 10^{-6}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$u_{1} \\leftarrow - {src}_{(1,1,-1)}^{22} - {src}_{(-1,1,-1)}^{21} - {src}_{(0,1,-1)}^{12} + {src}_{(-1,-1,-1)}^{19} - {src}_{(1,1,0)}^{9} - {src}_{(-1,1,0)}^{10} - {src}_{(0,1,0)}^{2} + {src}_{(-1,-1,0)}^{8} - {src}_{(1,1,1)}^{26} - {src}_{(-1,1,1)}^{25} - {src}_{(0,1,1)}^{16} + {src}_{(-1,-1,1)}^{23} + vel1Term$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$u_{2} \\leftarrow {src}_{(-1,1,-1)}^{21} + {src}_{(1,-1,-1)}^{20} + {src}_{(-1,-1,-1)}^{19} + {src}_{(0,-1,-1)}^{11} + {src}_{(-1,0,-1)}^{14} - {src}_{(1,0,1)}^{17} - {src}_{(1,1,1)}^{26} - {src}_{(-1,1,1)}^{25} - {src}_{(0,1,1)}^{16} - {src}_{(1,-1,1)}^{24} - {src}_{(-1,-1,1)}^{23} - {src}_{(0,-1,1)}^{15} - {src}_{(-1,0,1)}^{18} - {src}_{(0,0,1)}^{6} + vel2Term$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$\\sigma \\leftarrow \\sqrt{2 \\left(- {src}_{(1,0,-1)}^{13} - {src}_{(1,1,-1)}^{22} + {src}_{(-1,1,-1)}^{21} - {src}_{(1,-1,-1)}^{20} + {src}_{(-1,-1,-1)}^{19} + {src}_{(-1,0,-1)}^{14} + {src}_{(1,0,1)}^{17} + {src}_{(1,1,1)}^{26} - {src}_{(-1,1,1)}^{25} + {src}_{(1,-1,1)}^{24} - {src}_{(-1,-1,1)}^{23} - {src}_{(-1,0,1)}^{18} - u_{0} u_{2}\\right)^{2} + 2 \\left(- {src}_{(1,1,-1)}^{22} - {src}_{(-1,1,-1)}^{21} - {src}_{(0,1,-1)}^{12} + {src}_{(1,-1,-1)}^{20} + {src}_{(-1,-1,-1)}^{19} + {src}_{(0,-1,-1)}^{11} + {src}_{(1,1,1)}^{26} + {src}_{(-1,1,1)}^{25} + {src}_{(0,1,1)}^{16} - {src}_{(1,-1,1)}^{24} - {src}_{(-1,-1,1)}^{23} - {src}_{(0,-1,1)}^{15} - u_{1} u_{2}\\right)^{2} + 2 \\left({src}_{(1,1,-1)}^{22} - {src}_{(-1,1,-1)}^{21} - {src}_{(1,-1,-1)}^{20} + {src}_{(-1,-1,-1)}^{19} + {src}_{(1,1,0)}^{9} - {src}_{(-1,1,0)}^{10} - {src}_{(1,-1,0)}^{7} + {src}_{(-1,-1,0)}^{8} + {src}_{(1,1,1)}^{26} - {src}_{(-1,1,1)}^{25} - {src}_{(1,-1,1)}^{24} + {src}_{(-1,-1,1)}^{23} - u_{0} u_{1}\\right)^{2} + \\left({src}_{(1,0,-1)}^{13} + {src}_{(1,1,-1)}^{22} + {src}_{(-1,1,-1)}^{21} + {src}_{(0,1,-1)}^{12} + {src}_{(1,-1,-1)}^{20} + {src}_{(-1,-1,-1)}^{19} + {src}_{(0,-1,-1)}^{11} + {src}_{(-1,0,-1)}^{14} + {src}_{(0,0,-1)}^{5} + {src}_{(1,0,1)}^{17} + {src}_{(1,1,1)}^{26} + {src}_{(-1,1,1)}^{25} + {src}_{(0,1,1)}^{16} + {src}_{(1,-1,1)}^{24} + {src}_{(-1,-1,1)}^{23} + {src}_{(0,-1,1)}^{15} + {src}_{(-1,0,1)}^{18} + {src}_{(0,0,1)}^{6} - \\frac{\\delta_{\\rho}}{3} - u_{2}^{2}\\right)^{2} + \\left({src}_{(1,0,-1)}^{13} + {src}_{(1,1,-1)}^{22} + {src}_{(-1,1,-1)}^{21} + {src}_{(1,-1,-1)}^{20} + {src}_{(-1,-1,-1)}^{19} + {src}_{(-1,0,-1)}^{14} + {src}_{(1,0,0)}^{3} + {src}_{(1,1,0)}^{9} + {src}_{(-1,1,0)}^{10} + {src}_{(1,-1,0)}^{7} + {src}_{(-1,-1,0)}^{8} + {src}_{(1,0,1)}^{17} + {src}_{(1,1,1)}^{26} + {src}_{(-1,1,1)}^{25} + {src}_{(1,-1,1)}^{24} + {src}_{(-1,-1,1)}^{23} + {src}_{(-1,0,1)}^{18} + {src}_{(-1,0,0)}^{4} - \\frac{\\delta_{\\rho}}{3} - u_{0}^{2}\\right)^{2} + \\left({src}_{(1,1,-1)}^{22} + {src}_{(-1,1,-1)}^{21} + {src}_{(0,1,-1)}^{12} + {src}_{(1,-1,-1)}^{20} + {src}_{(-1,-1,-1)}^{19} + {src}_{(0,-1,-1)}^{11} + {src}_{(1,1,0)}^{9} + {src}_{(-1,1,0)}^{10} + {src}_{(0,1,0)}^{2} + {src}_{(1,-1,0)}^{7} + {src}_{(-1,-1,0)}^{8} + {src}_{(0,-1,0)}^{1} + {src}_{(1,1,1)}^{26} + {src}_{(-1,1,1)}^{25} + {src}_{(0,1,1)}^{16} + {src}_{(1,-1,1)}^{24} + {src}_{(-1,-1,1)}^{23} + {src}_{(0,-1,1)}^{15} - \\frac{\\delta_{\\rho}}{3} - u_{1}^{2}\\right)^{2}}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$\\xi_{7} \\leftarrow \\frac{1}{\\sigma}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$\\theta \\leftarrow 2.0 \\cdot 10^{-6} \\xi_{7}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$\\xi_{3} \\leftarrow \\frac{1}{1 - \\theta}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$rhs \\leftarrow \\xi_{3} \\left(\\sqrt{\\theta \\left(2 - \\theta\\right)} + 1\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$\\mu \\leftarrow \\frac{rhs^{2}}{6}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$\\tau \\leftarrow 3 \\mu + \\frac{1}{2}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$\\xi_{8} \\leftarrow \\frac{1}{\\tau}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$\\omega_{new} \\leftarrow \\begin{cases} 0.2 & \\text{for}\\: \\xi_{8} < 0.2 \\\\1.98 & \\text{for}\\: \\xi_{8} > 1.98 \\\\\\xi_{8} & \\text{otherwise} \\end{cases}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{0} \\leftarrow - 2.1680216802168 \\cdot 10^{-6} u_{0} \\cdot \\left(1 - \\frac{\\omega_{new}}{2}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{1} \\leftarrow - 5.42005420054201 \\cdot 10^{-7} u_{0} \\cdot \\left(1 - \\frac{\\omega_{new}}{2}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{2} \\leftarrow - 5.42005420054201 \\cdot 10^{-7} u_{0} \\cdot \\left(1 - \\frac{\\omega_{new}}{2}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{3} \\leftarrow \\left(1 - \\frac{\\omega_{new}}{2}\\right) \\left(1.0840108401084 \\cdot 10^{-6} u_{0} - 5.42005420054201 \\cdot 10^{-7}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{4} \\leftarrow \\left(1 - \\frac{\\omega_{new}}{2}\\right) \\left(1.0840108401084 \\cdot 10^{-6} u_{0} + 5.42005420054201 \\cdot 10^{-7}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{5} \\leftarrow - 5.42005420054201 \\cdot 10^{-7} u_{0} \\cdot \\left(1 - \\frac{\\omega_{new}}{2}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{6} \\leftarrow - 5.42005420054201 \\cdot 10^{-7} u_{0} \\cdot \\left(1 - \\frac{\\omega_{new}}{2}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{7} \\leftarrow \\left(1 - \\frac{\\omega_{new}}{2}\\right) \\left(2.710027100271 \\cdot 10^{-7} u_{0} - 4.0650406504065 \\cdot 10^{-7} u_{1} - 1.3550135501355 \\cdot 10^{-7}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{8} \\leftarrow \\left(1 - \\frac{\\omega_{new}}{2}\\right) \\left(2.710027100271 \\cdot 10^{-7} u_{0} + 4.0650406504065 \\cdot 10^{-7} u_{1} + 1.3550135501355 \\cdot 10^{-7}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{9} \\leftarrow \\left(1 - \\frac{\\omega_{new}}{2}\\right) \\left(2.710027100271 \\cdot 10^{-7} u_{0} + 4.0650406504065 \\cdot 10^{-7} u_{1} - 1.3550135501355 \\cdot 10^{-7}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{10} \\leftarrow \\left(1 - \\frac{\\omega_{new}}{2}\\right) \\left(2.710027100271 \\cdot 10^{-7} u_{0} - 4.0650406504065 \\cdot 10^{-7} u_{1} + 1.3550135501355 \\cdot 10^{-7}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{11} \\leftarrow - 1.3550135501355 \\cdot 10^{-7} u_{0} \\cdot \\left(1 - \\frac{\\omega_{new}}{2}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{12} \\leftarrow - 1.3550135501355 \\cdot 10^{-7} u_{0} \\cdot \\left(1 - \\frac{\\omega_{new}}{2}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{13} \\leftarrow \\left(1 - \\frac{\\omega_{new}}{2}\\right) \\left(2.710027100271 \\cdot 10^{-7} u_{0} - 4.0650406504065 \\cdot 10^{-7} u_{2} - 1.3550135501355 \\cdot 10^{-7}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{14} \\leftarrow \\left(1 - \\frac{\\omega_{new}}{2}\\right) \\left(2.710027100271 \\cdot 10^{-7} u_{0} + 4.0650406504065 \\cdot 10^{-7} u_{2} + 1.3550135501355 \\cdot 10^{-7}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{15} \\leftarrow - 1.3550135501355 \\cdot 10^{-7} u_{0} \\cdot \\left(1 - \\frac{\\omega_{new}}{2}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{16} \\leftarrow - 1.3550135501355 \\cdot 10^{-7} u_{0} \\cdot \\left(1 - \\frac{\\omega_{new}}{2}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{17} \\leftarrow \\left(1 - \\frac{\\omega_{new}}{2}\\right) \\left(2.710027100271 \\cdot 10^{-7} u_{0} + 4.0650406504065 \\cdot 10^{-7} u_{2} - 1.3550135501355 \\cdot 10^{-7}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{18} \\leftarrow \\left(1 - \\frac{\\omega_{new}}{2}\\right) \\left(2.710027100271 \\cdot 10^{-7} u_{0} - 4.0650406504065 \\cdot 10^{-7} u_{2} + 1.3550135501355 \\cdot 10^{-7}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{19} \\leftarrow \\left(1 - \\frac{\\omega_{new}}{2}\\right) \\left(6.77506775067751 \\cdot 10^{-8} u_{0} + 1.01626016260163 \\cdot 10^{-7} u_{1} + 1.01626016260163 \\cdot 10^{-7} u_{2} + 3.38753387533875 \\cdot 10^{-8}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{20} \\leftarrow \\left(1 - \\frac{\\omega_{new}}{2}\\right) \\left(6.77506775067751 \\cdot 10^{-8} u_{0} - 1.01626016260163 \\cdot 10^{-7} u_{1} - 1.01626016260163 \\cdot 10^{-7} u_{2} - 3.38753387533875 \\cdot 10^{-8}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{21} \\leftarrow \\left(1 - \\frac{\\omega_{new}}{2}\\right) \\left(6.77506775067751 \\cdot 10^{-8} u_{0} - 1.01626016260163 \\cdot 10^{-7} u_{1} + 1.01626016260163 \\cdot 10^{-7} u_{2} + 3.38753387533875 \\cdot 10^{-8}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{22} \\leftarrow \\left(1 - \\frac{\\omega_{new}}{2}\\right) \\left(6.77506775067751 \\cdot 10^{-8} u_{0} + 1.01626016260163 \\cdot 10^{-7} u_{1} - 1.01626016260163 \\cdot 10^{-7} u_{2} - 3.38753387533875 \\cdot 10^{-8}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{23} \\leftarrow \\left(1 - \\frac{\\omega_{new}}{2}\\right) \\left(6.77506775067751 \\cdot 10^{-8} u_{0} + 1.01626016260163 \\cdot 10^{-7} u_{1} - 1.01626016260163 \\cdot 10^{-7} u_{2} + 3.38753387533875 \\cdot 10^{-8}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{24} \\leftarrow \\left(1 - \\frac{\\omega_{new}}{2}\\right) \\left(6.77506775067751 \\cdot 10^{-8} u_{0} - 1.01626016260163 \\cdot 10^{-7} u_{1} + 1.01626016260163 \\cdot 10^{-7} u_{2} - 3.38753387533875 \\cdot 10^{-8}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{25} \\leftarrow \\left(1 - \\frac{\\omega_{new}}{2}\\right) \\left(6.77506775067751 \\cdot 10^{-8} u_{0} - 1.01626016260163 \\cdot 10^{-7} u_{1} - 1.01626016260163 \\cdot 10^{-7} u_{2} + 3.38753387533875 \\cdot 10^{-8}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$forceTerm_{26} \\leftarrow \\left(1 - \\frac{\\omega_{new}}{2}\\right) \\left(6.77506775067751 \\cdot 10^{-8} u_{0} + 1.01626016260163 \\cdot 10^{-7} u_{1} + 1.01626016260163 \\cdot 10^{-7} u_{2} - 3.38753387533875 \\cdot 10^{-8}\\right)$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$momdensity_{0} \\leftarrow - {src}_{(1,0,-1)}^{13} - {src}_{(1,1,-1)}^{22} - {src}_{(1,-1,-1)}^{20} - {src}_{(1,0,0)}^{3} - {src}_{(1,1,0)}^{9} - {src}_{(1,-1,0)}^{7} - {src}_{(1,0,1)}^{17} - {src}_{(1,1,1)}^{26} - {src}_{(1,-1,1)}^{24} + vel0Term$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$momdensity_{1} \\leftarrow - {src}_{(1,1,-1)}^{22} - {src}_{(-1,1,-1)}^{21} - {src}_{(0,1,-1)}^{12} + {src}_{(-1,-1,-1)}^{19} - {src}_{(1,1,0)}^{9} - {src}_{(-1,1,0)}^{10} - {src}_{(0,1,0)}^{2} + {src}_{(-1,-1,0)}^{8} - {src}_{(1,1,1)}^{26} - {src}_{(-1,1,1)}^{25} - {src}_{(0,1,1)}^{16} + {src}_{(-1,-1,1)}^{23} + vel1Term$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$momdensity_{2} \\leftarrow {src}_{(-1,1,-1)}^{21} + {src}_{(1,-1,-1)}^{20} + {src}_{(-1,-1,-1)}^{19} + {src}_{(0,-1,-1)}^{11} + {src}_{(-1,0,-1)}^{14} - {src}_{(1,0,1)}^{17} - {src}_{(1,1,1)}^{26} - {src}_{(-1,1,1)}^{25} - {src}_{(0,1,1)}^{16} - {src}_{(1,-1,1)}^{24} - {src}_{(-1,-1,1)}^{23} - {src}_{(0,-1,1)}^{15} - {src}_{(-1,0,1)}^{18} - {src}_{(0,0,1)}^{6} + vel2Term$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$\\xi_{0} \\leftarrow momdensity_{0} + 1.21951219512195 \\cdot 10^{-6}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$u0Mu1 \\leftarrow u_{0} - u_{1}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$u0Pu1 \\leftarrow u_{0} + u_{1}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$u1Pu2 \\leftarrow u_{1} + u_{2}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$u1Mu2 \\leftarrow u_{1} - u_{2}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$u0Mu2 \\leftarrow u_{0} - u_{2}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$$u0Pu2 \\leftarrow u_{0} + u_{2}$$</td>  </tr> </table><div>Main Assignments:</div><table style=\"border:none; width: 100%; \"><tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{0} \\leftarrow - {src}_{(0,0,0)}^{0} \\omega_{new} + {src}_{(0,0,0)}^{0} + \\frac{8 \\delta_{\\rho} \\omega_{new}}{27} + forceTerm_{0} - \\frac{4 \\omega_{new} u_{0}^{2}}{9} - \\frac{4 \\omega_{new} u_{1}^{2}}{9} - \\frac{4 \\omega_{new} u_{2}^{2}}{9}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{1} \\leftarrow - {src}_{(0,-1,0)}^{1} \\omega_{new} + {src}_{(0,-1,0)}^{1} + \\frac{2 \\delta_{\\rho} \\omega_{new}}{27} + forceTerm_{1} - \\frac{\\omega_{new} u_{0}^{2}}{9} + \\frac{2 \\omega_{new} u_{1}^{2}}{9} + \\frac{2 \\omega_{new} u_{1}}{9} - \\frac{\\omega_{new} u_{2}^{2}}{9}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{2} \\leftarrow - {src}_{(0,1,0)}^{2} \\omega_{new} + {src}_{(0,1,0)}^{2} + \\frac{2 \\delta_{\\rho} \\omega_{new}}{27} + forceTerm_{2} - \\frac{\\omega_{new} u_{0}^{2}}{9} + \\frac{2 \\omega_{new} u_{1}^{2}}{9} - \\frac{2 \\omega_{new} u_{1}}{9} - \\frac{\\omega_{new} u_{2}^{2}}{9}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{3} \\leftarrow - {src}_{(1,0,0)}^{3} \\omega_{new} + {src}_{(1,0,0)}^{3} + \\frac{2 \\delta_{\\rho} \\omega_{new}}{27} + forceTerm_{3} + \\frac{2 \\omega_{new} u_{0}^{2}}{9} - \\frac{2 \\omega_{new} u_{0}}{9} - \\frac{\\omega_{new} u_{1}^{2}}{9} - \\frac{\\omega_{new} u_{2}^{2}}{9}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{4} \\leftarrow - {src}_{(-1,0,0)}^{4} \\omega_{new} + {src}_{(-1,0,0)}^{4} + \\frac{2 \\delta_{\\rho} \\omega_{new}}{27} + forceTerm_{4} + \\frac{2 \\omega_{new} u_{0}^{2}}{9} + \\frac{2 \\omega_{new} u_{0}}{9} - \\frac{\\omega_{new} u_{1}^{2}}{9} - \\frac{\\omega_{new} u_{2}^{2}}{9}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{5} \\leftarrow - {src}_{(0,0,-1)}^{5} \\omega_{new} + {src}_{(0,0,-1)}^{5} + \\frac{2 \\delta_{\\rho} \\omega_{new}}{27} + forceTerm_{5} - \\frac{\\omega_{new} u_{0}^{2}}{9} - \\frac{\\omega_{new} u_{1}^{2}}{9} + \\frac{2 \\omega_{new} u_{2}^{2}}{9} + \\frac{2 \\omega_{new} u_{2}}{9}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{6} \\leftarrow - {src}_{(0,0,1)}^{6} \\omega_{new} + {src}_{(0,0,1)}^{6} + \\frac{2 \\delta_{\\rho} \\omega_{new}}{27} + forceTerm_{6} - \\frac{\\omega_{new} u_{0}^{2}}{9} - \\frac{\\omega_{new} u_{1}^{2}}{9} + \\frac{2 \\omega_{new} u_{2}^{2}}{9} - \\frac{2 \\omega_{new} u_{2}}{9}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{7} \\leftarrow - {src}_{(1,-1,0)}^{7} \\omega_{new} + {src}_{(1,-1,0)}^{7} + \\frac{\\delta_{\\rho} \\omega_{new}}{54} + forceTerm_{7} + \\frac{\\omega_{new} u0Mu1^{2}}{12} - \\frac{\\omega_{new} u_{0}^{2}}{36} - \\frac{\\omega_{new} u_{0}}{18} - \\frac{\\omega_{new} u_{1}^{2}}{36} + \\frac{\\omega_{new} u_{1}}{18} - \\frac{\\omega_{new} u_{2}^{2}}{36}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{8} \\leftarrow - {src}_{(-1,-1,0)}^{8} \\omega_{new} + {src}_{(-1,-1,0)}^{8} + \\frac{\\delta_{\\rho} \\omega_{new}}{54} + forceTerm_{8} + \\frac{\\omega_{new} u0Pu1^{2}}{12} - \\frac{\\omega_{new} u_{0}^{2}}{36} + \\frac{\\omega_{new} u_{0}}{18} - \\frac{\\omega_{new} u_{1}^{2}}{36} + \\frac{\\omega_{new} u_{1}}{18} - \\frac{\\omega_{new} u_{2}^{2}}{36}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{9} \\leftarrow - {src}_{(1,1,0)}^{9} \\omega_{new} + {src}_{(1,1,0)}^{9} + \\frac{\\delta_{\\rho} \\omega_{new}}{54} + forceTerm_{9} + \\frac{\\omega_{new} u0Pu1^{2}}{12} - \\frac{\\omega_{new} u_{0}^{2}}{36} - \\frac{\\omega_{new} u_{0}}{18} - \\frac{\\omega_{new} u_{1}^{2}}{36} - \\frac{\\omega_{new} u_{1}}{18} - \\frac{\\omega_{new} u_{2}^{2}}{36}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{10} \\leftarrow - {src}_{(-1,1,0)}^{10} \\omega_{new} + {src}_{(-1,1,0)}^{10} + \\frac{\\delta_{\\rho} \\omega_{new}}{54} + forceTerm_{10} + \\frac{\\omega_{new} u0Mu1^{2}}{12} - \\frac{\\omega_{new} u_{0}^{2}}{36} + \\frac{\\omega_{new} u_{0}}{18} - \\frac{\\omega_{new} u_{1}^{2}}{36} - \\frac{\\omega_{new} u_{1}}{18} - \\frac{\\omega_{new} u_{2}^{2}}{36}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{11} \\leftarrow - {src}_{(0,-1,-1)}^{11} \\omega_{new} + {src}_{(0,-1,-1)}^{11} + \\frac{\\delta_{\\rho} \\omega_{new}}{54} + forceTerm_{11} + \\frac{\\omega_{new} u1Pu2^{2}}{12} - \\frac{\\omega_{new} u_{0}^{2}}{36} - \\frac{\\omega_{new} u_{1}^{2}}{36} + \\frac{\\omega_{new} u_{1}}{18} - \\frac{\\omega_{new} u_{2}^{2}}{36} + \\frac{\\omega_{new} u_{2}}{18}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{12} \\leftarrow - {src}_{(0,1,-1)}^{12} \\omega_{new} + {src}_{(0,1,-1)}^{12} + \\frac{\\delta_{\\rho} \\omega_{new}}{54} + forceTerm_{12} + \\frac{\\omega_{new} u1Mu2^{2}}{12} - \\frac{\\omega_{new} u_{0}^{2}}{36} - \\frac{\\omega_{new} u_{1}^{2}}{36} - \\frac{\\omega_{new} u_{1}}{18} - \\frac{\\omega_{new} u_{2}^{2}}{36} + \\frac{\\omega_{new} u_{2}}{18}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{13} \\leftarrow - {src}_{(1,0,-1)}^{13} \\omega_{new} + {src}_{(1,0,-1)}^{13} + \\frac{\\delta_{\\rho} \\omega_{new}}{54} + forceTerm_{13} + \\frac{\\omega_{new} u0Mu2^{2}}{12} - \\frac{\\omega_{new} u_{0}^{2}}{36} - \\frac{\\omega_{new} u_{0}}{18} - \\frac{\\omega_{new} u_{1}^{2}}{36} - \\frac{\\omega_{new} u_{2}^{2}}{36} + \\frac{\\omega_{new} u_{2}}{18}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{14} \\leftarrow - {src}_{(-1,0,-1)}^{14} \\omega_{new} + {src}_{(-1,0,-1)}^{14} + \\frac{\\delta_{\\rho} \\omega_{new}}{54} + forceTerm_{14} + \\frac{\\omega_{new} u0Pu2^{2}}{12} - \\frac{\\omega_{new} u_{0}^{2}}{36} + \\frac{\\omega_{new} u_{0}}{18} - \\frac{\\omega_{new} u_{1}^{2}}{36} - \\frac{\\omega_{new} u_{2}^{2}}{36} + \\frac{\\omega_{new} u_{2}}{18}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{15} \\leftarrow - {src}_{(0,-1,1)}^{15} \\omega_{new} + {src}_{(0,-1,1)}^{15} + \\frac{\\delta_{\\rho} \\omega_{new}}{54} + forceTerm_{15} + \\frac{\\omega_{new} u1Mu2^{2}}{12} - \\frac{\\omega_{new} u_{0}^{2}}{36} - \\frac{\\omega_{new} u_{1}^{2}}{36} + \\frac{\\omega_{new} u_{1}}{18} - \\frac{\\omega_{new} u_{2}^{2}}{36} - \\frac{\\omega_{new} u_{2}}{18}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{16} \\leftarrow - {src}_{(0,1,1)}^{16} \\omega_{new} + {src}_{(0,1,1)}^{16} + \\frac{\\delta_{\\rho} \\omega_{new}}{54} + forceTerm_{16} + \\frac{\\omega_{new} u1Pu2^{2}}{12} - \\frac{\\omega_{new} u_{0}^{2}}{36} - \\frac{\\omega_{new} u_{1}^{2}}{36} - \\frac{\\omega_{new} u_{1}}{18} - \\frac{\\omega_{new} u_{2}^{2}}{36} - \\frac{\\omega_{new} u_{2}}{18}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{17} \\leftarrow - {src}_{(1,0,1)}^{17} \\omega_{new} + {src}_{(1,0,1)}^{17} + \\frac{\\delta_{\\rho} \\omega_{new}}{54} + forceTerm_{17} + \\frac{\\omega_{new} u0Pu2^{2}}{12} - \\frac{\\omega_{new} u_{0}^{2}}{36} - \\frac{\\omega_{new} u_{0}}{18} - \\frac{\\omega_{new} u_{1}^{2}}{36} - \\frac{\\omega_{new} u_{2}^{2}}{36} - \\frac{\\omega_{new} u_{2}}{18}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{18} \\leftarrow - {src}_{(-1,0,1)}^{18} \\omega_{new} + {src}_{(-1,0,1)}^{18} + \\frac{\\delta_{\\rho} \\omega_{new}}{54} + forceTerm_{18} + \\frac{\\omega_{new} u0Mu2^{2}}{12} - \\frac{\\omega_{new} u_{0}^{2}}{36} + \\frac{\\omega_{new} u_{0}}{18} - \\frac{\\omega_{new} u_{1}^{2}}{36} - \\frac{\\omega_{new} u_{2}^{2}}{36} - \\frac{\\omega_{new} u_{2}}{18}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{19} \\leftarrow - {src}_{(-1,-1,-1)}^{19} \\omega_{new} + {src}_{(-1,-1,-1)}^{19} + \\frac{\\delta_{\\rho} \\omega_{new}}{216} + forceTerm_{19} + \\frac{\\omega_{new} u0Pu1^{2}}{48} + \\frac{\\omega_{new} u0Pu2^{2}}{48} + \\frac{\\omega_{new} u1Pu2^{2}}{48} - \\frac{\\omega_{new} u_{0}^{2}}{36} + \\frac{\\omega_{new} u_{0}}{72} - \\frac{\\omega_{new} u_{1}^{2}}{36} + \\frac{\\omega_{new} u_{1}}{72} - \\frac{\\omega_{new} u_{2}^{2}}{36} + \\frac{\\omega_{new} u_{2}}{72}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{20} \\leftarrow - {src}_{(1,-1,-1)}^{20} \\omega_{new} + {src}_{(1,-1,-1)}^{20} + \\frac{\\delta_{\\rho} \\omega_{new}}{216} + forceTerm_{20} + \\frac{\\omega_{new} u0Mu1^{2}}{48} + \\frac{\\omega_{new} u0Mu2^{2}}{48} + \\frac{\\omega_{new} u1Pu2^{2}}{48} - \\frac{\\omega_{new} u_{0}^{2}}{36} - \\frac{\\omega_{new} u_{0}}{72} - \\frac{\\omega_{new} u_{1}^{2}}{36} + \\frac{\\omega_{new} u_{1}}{72} - \\frac{\\omega_{new} u_{2}^{2}}{36} + \\frac{\\omega_{new} u_{2}}{72}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{21} \\leftarrow - {src}_{(-1,1,-1)}^{21} \\omega_{new} + {src}_{(-1,1,-1)}^{21} + \\frac{\\delta_{\\rho} \\omega_{new}}{216} + forceTerm_{21} + \\frac{\\omega_{new} u0Mu1^{2}}{48} + \\frac{\\omega_{new} u0Pu2^{2}}{48} + \\frac{\\omega_{new} u1Mu2^{2}}{48} - \\frac{\\omega_{new} u_{0}^{2}}{36} + \\frac{\\omega_{new} u_{0}}{72} - \\frac{\\omega_{new} u_{1}^{2}}{36} - \\frac{\\omega_{new} u_{1}}{72} - \\frac{\\omega_{new} u_{2}^{2}}{36} + \\frac{\\omega_{new} u_{2}}{72}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{22} \\leftarrow - {src}_{(1,1,-1)}^{22} \\omega_{new} + {src}_{(1,1,-1)}^{22} + \\frac{\\delta_{\\rho} \\omega_{new}}{216} + forceTerm_{22} + \\frac{\\omega_{new} u0Mu2^{2}}{48} + \\frac{\\omega_{new} u0Pu1^{2}}{48} + \\frac{\\omega_{new} u1Mu2^{2}}{48} - \\frac{\\omega_{new} u_{0}^{2}}{36} - \\frac{\\omega_{new} u_{0}}{72} - \\frac{\\omega_{new} u_{1}^{2}}{36} - \\frac{\\omega_{new} u_{1}}{72} - \\frac{\\omega_{new} u_{2}^{2}}{36} + \\frac{\\omega_{new} u_{2}}{72}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{23} \\leftarrow - {src}_{(-1,-1,1)}^{23} \\omega_{new} + {src}_{(-1,-1,1)}^{23} + \\frac{\\delta_{\\rho} \\omega_{new}}{216} + forceTerm_{23} + \\frac{\\omega_{new} u0Mu2^{2}}{48} + \\frac{\\omega_{new} u0Pu1^{2}}{48} + \\frac{\\omega_{new} u1Mu2^{2}}{48} - \\frac{\\omega_{new} u_{0}^{2}}{36} + \\frac{\\omega_{new} u_{0}}{72} - \\frac{\\omega_{new} u_{1}^{2}}{36} + \\frac{\\omega_{new} u_{1}}{72} - \\frac{\\omega_{new} u_{2}^{2}}{36} - \\frac{\\omega_{new} u_{2}}{72}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{24} \\leftarrow - {src}_{(1,-1,1)}^{24} \\omega_{new} + {src}_{(1,-1,1)}^{24} + \\frac{\\delta_{\\rho} \\omega_{new}}{216} + forceTerm_{24} + \\frac{\\omega_{new} u0Mu1^{2}}{48} + \\frac{\\omega_{new} u0Pu2^{2}}{48} + \\frac{\\omega_{new} u1Mu2^{2}}{48} - \\frac{\\omega_{new} u_{0}^{2}}{36} - \\frac{\\omega_{new} u_{0}}{72} - \\frac{\\omega_{new} u_{1}^{2}}{36} + \\frac{\\omega_{new} u_{1}}{72} - \\frac{\\omega_{new} u_{2}^{2}}{36} - \\frac{\\omega_{new} u_{2}}{72}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{25} \\leftarrow - {src}_{(-1,1,1)}^{25} \\omega_{new} + {src}_{(-1,1,1)}^{25} + \\frac{\\delta_{\\rho} \\omega_{new}}{216} + forceTerm_{25} + \\frac{\\omega_{new} u0Mu1^{2}}{48} + \\frac{\\omega_{new} u0Mu2^{2}}{48} + \\frac{\\omega_{new} u1Pu2^{2}}{48} - \\frac{\\omega_{new} u_{0}^{2}}{36} + \\frac{\\omega_{new} u_{0}}{72} - \\frac{\\omega_{new} u_{1}^{2}}{36} - \\frac{\\omega_{new} u_{1}}{72} - \\frac{\\omega_{new} u_{2}^{2}}{36} - \\frac{\\omega_{new} u_{2}}{72}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${dst}_{(0,0,0)}^{26} \\leftarrow - {src}_{(1,1,1)}^{26} \\omega_{new} + {src}_{(1,1,1)}^{26} + \\frac{\\delta_{\\rho} \\omega_{new}}{216} + forceTerm_{26} + \\frac{\\omega_{new} u0Pu1^{2}}{48} + \\frac{\\omega_{new} u0Pu2^{2}}{48} + \\frac{\\omega_{new} u1Pu2^{2}}{48} - \\frac{\\omega_{new} u_{0}^{2}}{36} - \\frac{\\omega_{new} u_{0}}{72} - \\frac{\\omega_{new} u_{1}^{2}}{36} - \\frac{\\omega_{new} u_{1}}{72} - \\frac{\\omega_{new} u_{2}^{2}}{36} - \\frac{\\omega_{new} u_{2}}{72}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${velField}_{(0,0,0)}^{0} \\leftarrow \\xi_{0}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${velField}_{(0,0,0)}^{1} \\leftarrow momdensity_{1}$$</td>  </tr> <tr style=\"border:none\"> <td style=\"border:none\">$${velField}_{(0,0,0)}^{2} \\leftarrow momdensity_{2}$$</td>  </tr> </table>"
-      ],
-      "text/plain": [
-       "AssignmentCollection: dst_C^10, dst_C^24, dst_C^4, dst_C^26, dst_C^7, velField_C^1, dst_C^17, dst_C^6, dst_C^25, velField_C^2, dst_C^8, dst_C^19, dst_C^12, dst_C^21, dst_C^14, dst_C^18, velField_C^0, dst_C^9, dst_C^16, dst_C^3, dst_C^5, dst_C^2, dst_C^1, dst_C^11, dst_C^20, dst_C^22, dst_C^13, dst_C^15, dst_C^0, dst_C^23 <- f(src_BE^13, src_TE^17, src_T^6, src_BSE^20, src_TW^18, src_E^3, src_TNE^26, src_TSW^23, src_BW^14, src_BNE^22, src_BSW^19, src_C^0, src_N^2, src_S^1, src_TS^15, src_NE^9, src_B^5, src_SE^7, src_TN^16, src_TNW^25, src_W^4, src_SW^8, src_NW^10, src_TSE^24, src_BS^11, src_BN^12, src_BNW^21)"
-      ]
-     },
-     "execution_count": 8,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
+   "outputs": [],
    "source": [
-    "update"
+    "# you can look at the complete equation set by removing the comment below\n",
+    "\n",
+    "# update"
    ]
   },
   {
@@ -238,7 +226,7 @@
     {
      "data": {
       "text/plain": [
-       "<matplotlib.colorbar.Colorbar at 0x130dec1f0>"
+       "<matplotlib.colorbar.Colorbar at 0x12dd99400>"
       ]
      },
      "execution_count": 12,
@@ -303,7 +291,7 @@
     {
      "data": {
       "text/plain": [
-       "<matplotlib.colorbar.Colorbar at 0x163128160>"
+       "<matplotlib.colorbar.Colorbar at 0x12f07a610>"
       ]
      },
      "execution_count": 14,
@@ -337,7 +325,7 @@
     {
      "data": {
       "text/plain": [
-       "<matplotlib.legend.Legend at 0x1631c7fd0>"
+       "<matplotlib.legend.Legend at 0x12f026a30>"
       ]
      },
      "execution_count": 15,
diff --git a/lbmpy/non_newtonian_models.py b/lbmpy/non_newtonian_models.py
index 2ab4392ee4c03592874dde3cd5125274138a2c46..40549eb0e6ea9bbae09e238cfae5a1e9c1a55527 100644
--- a/lbmpy/non_newtonian_models.py
+++ b/lbmpy/non_newtonian_models.py
@@ -58,7 +58,7 @@ def add_cassons_model(collision_rule, parameter: CassonsParameters, omega_output
 
     # rhs of equation 14 in https://doi.org/10.1007/s10955-005-8415-x
     # Note that C_2 / C_4 = 3 for all configurations thus we directly insert it here
-    eq14 = one / (one - theta) * (one + sp.sqrt(theta * (one + rho / eta * sp.Rational(3, 2) * (one - theta))))
+    eq14 = one / (one - theta) * (one + sp.sqrt(theta * (one + rho / eta * sp.Rational(1, 6) * (one - theta))))
 
     new_omega = one / tau
     omega_cond = sp.Piecewise((omega_min, new_omega < omega_min), (omega_max, new_omega > omega_max), (new_omega, True))