clean up the example a little