Adding the second part of the assignment