Python Fit Function To Graph