Added basic code to handle upgrades (need to make sure it isnt C+P though)
Added graphics