Update image tree generation

Update image tree generation for generate color based on element
categories
Merge branch 'master' of https://github.com/Dzemo/igmt