Working 2D fluid demo based on the viscoelastic simulation paper