Update DrawFlux.m