Added training entire network