Fixed Point Iteration Python