Improvements to training

Signed-off-by: MaloW <MaloW@spray.se>