Most of training UI done