updated main - added TreeMaps