Changing cpp file little bit