[updated] Color coding for types are working, legends are added