EZ does it! Extensions of the EZ-diffusion model