cleaned up a bit and included libraries for plotting and complex numbers