Training with branches again