Added unit upgrade feature (updated)