Modified to output/plot real and imaginary components separately (instead of absolute value)