Also I should just write this in Cython since I do a lot of unnecessary stuff when using numpy.