Clean up some of the example code.