Added color legend (closes #3)