added testing for teacher model