Improve the vector field plot