Made some changes based off of assignment screenshot.