work from week5 class