cleanup
fix a bug in y axis diffusion. thank you @githole!
remove duplicate in README
update screenshot