if (color == 2)
                        {
                            g2.setColor(Color.yellow);
                        }
                        else
                        {
                            g2.setColor(Color.red);
                        }