part 2/4 of course project 1