Fixing figure legend