Finishing the second part of the challenge