Rework level up dialog to behave more like vanilla