Add software upgrade feature