Adding legend and clarity