Added new code to the goal 7 assignment