Code updated to shipping version of book.