Made changes following a Youtube video.