Fix visualization & take out allocations / deallocations in the algorithms.
Add Crank-Nicolson & some minor reorganization.