added a line of code per the homework