Colour according to nidis instead of nids in spike_pos.py