Commented and created a colour legend