Changed some bits of code while watching solution video.