Add code to the example