minor changes to example in unit6