Training controller needs  to be completely changed