Added unit upgrade feature