Adding in log loss check function