first version problem 2
Finished problem2
finished problem2