Follow the instructions to modify this piece of code