Title, legend and other UI improvements