Added a function for returning the color