part 4/4 of course project 1