Add upgrade support in level