Python script for a linear fitter on data