Flag didn't work. Creates a new class for the cross entropy neural network