changed colours into an array with less code