Add safe upgrade