Fixed Point Iteration In Python