Added 'mean_square_error' function