Added WF figure, code