Added tests, but getting complex numbers.