Use linspace instead of arange (simpler syntax and yields the same)
Make the calculation of I'_S a bit clearer in P2-8