First problem complete
Completed problem 2