Update CreateUpgrade with color support