Modified lagrange_rho function to use D[1-4] for flow direction. 
Now it looks like it's working O_o