Minor README tweaks.
Added gradient.
Added some visualization code.