fixing plotting bugs, exp(imaginary fun)