Final version

Lots of clean-up and prettification, changed color mapping to be
diverging color scale