Added a few changes where each triangle now has a color id and the display class decides which color that corresponds to