clean up example a bit