Adds flag for specifiying whether to use cross entropy or square loss
function.