def loss_function(X, y, w, b):