fixes
End. Still really slow to converge, but not sure what to do about it