Add KBART class functionality