After the modification of the code