Fix legend, remove unwanted functions
Merge pull request #8 from cjeast/cjeast/fix-legend

Fix legend, remove unwanted functions