Update header and cpp file