Choosing colour based on array