Hard coded legend - need to fix