Working to add legend