adding some lines to have this same code