Remove non used Training Class, fix issue when no style is provided