refactor prob/transform