This worked well for me too. However, i would also like to include the legend with color codes but im struggling to do so, anyone know a workaround?