Add API to remove and add legend
Add more docs