A few improvements i'd like to suggest:
- Give the player a choice whether to change the weapon for a new one or not. Picking up chests is complete gambling and often you end up with a weapon worse than you had previously.
- Maybe add a few chords into the music (I recommend looking up some basic music theory)
- Give the player a choice which upgrades to make