clean up example a little