Update print function to take a color.