Add ifs and color