Re-add colors for different domains
Comment on different position
More scalable way of adding different colours

TODO: refactor this crap