Added test for LogSquared and fixed Loss