Update wk3_magnitude.m