Fixed little bugs.
Changed to numpy sums. SO MUCH FASTER