added input for color