Some changes and bugs Fixed - Jeremy Kim

Now takes U vector which is basically your initial guess vector and now
the function will do a while loop instead of a for loop. When the
convergence test percentage is less than 1%, it will break out of the
while loop.