Add tentative solution to colour problem by defining two templates