Minor correction to legend and comments
Minor correction to legend and comments