setColor added small fix