add training diagrams with better contrast