Make setting colors for plot.neuron more consistent with plot3d.neuron

Implements #55.