Add legend. Fix #5