Fixed log-legend and made colormap a bit nicer.
Fixed log-shading of surface.
Hopefully removed source of flickering.