Addition for stretchgoal 3