Edit train colours