Finish class5 exercise 2