add color variation to display function