added wk plotting