update init to grab everything from diffusion
add stokes diffusion function to estimate diffusion in correct units
clean up load_timestep, improve timestep list and other bits