Finish class 4 exercise 1