Adding the the first part from the second task