finished implementing unit training system