ENH: color by dx