Added upgrade option