Tweaked level up code