Add variable color scale, make legend dynamic