Added the color parameter functionality.