cleaned up atom tracing in diffusion interactives