minor changes to CNN