Changes after code walkthrough.