various little edits to diffusion preprocessing code

I made some changes to allow more flexibility with diffusion preprocessing. The changes shouldn't impact standard workflows (I hope!).