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

A start towards an implementation of #55.