Change code to match instructor solution - still need to complete steps 4 and 5.