More hit plots

Another plot that colors the neurons based on labels. Also added a hack
to prevent legend from being plotted outside the visible area.