add wk2 code follow through