add parameter for setting the color in the functions for drawing lines