complete wk2 assignment