bug fix with color additions