Made changes to complete Part 4 of the tutorial