Polynomial fit in python