Increase the max_iter to 1000