Update trainer.java

Update