Fit A Function To Data Python