updates for w2 lab6
updates for w2 lab6