Adding part 5 of week 3