update legend to make colors consistent