part2 modified advection

out[22] is ut + aux and ready for putting in latex