Added edge colour prediction.