Add title and improvde legend