adding a legend to the squares visualization and attemp to correct labels, both not working