Added upgrade rules