Adding worklist functionality