Fixing final example