work from week4's class