minor edit of week 4 lab assignment