enhance apply2trace kwarg
refactor, add docs