wk4_4 partial solution with bug
wk4_4 still has bug