Intermediate commit: Try optimizing script.

Still is problematic -- the solution is probability > 1!!

(Not sure why.)
Add debug method
Clean up run_optimization script
Try not recalculating Xi-Xj over and over again

This will hopefully speed things up, at least a little bit.

Could actually save the 'u' values in grad for even greater savings.
Add values for theta, for now