Added rainbow upgrade support. More coming soon :)