clean the code and add folders tree feature.