minor changes to cross-entropy toward framework