add color code for tpus