Made code match example.