Changes to example solution