Modify files to habdle a separate Training class and running only object class prediction test in a separate executable