Using a loss function for simple regression