Add more visualization code.
Spiral initial conditions + tests.