Fixed Point Iteration Python Code