Made some changes to program after watching solution videos.