Adds legend improvements