add folder /tree for train result, add /dataset for grouping Dataset classes together, edit source code to use as described, fully use main json file
little change on main.py