curve fitting problems with python