Clean up example a little more.