Added color key system for #21

Will make the colours a little nicer at some point and maybe filter via clicking but this'll do for the moment.