Add function to make upgrade easier