adding nn training