Added upgrade system