Fixed some errors, guess I have to keep the imaginary part.