Update to work with legend = F argument
Update to work with legend = F argument